train validation and test data sets using python